#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <functional>
#include "../Common/Log.hpp"
#include "../Common/thread.hpp"
#include "../Common/Cond.hpp"
#include "../Common/Mutex.hpp"

namespace ThreadPoolModule
{
    using namespace ThreadModlue;
    using namespace LogModule;
    using namespace CondModule;
    using namespace MutexModule;

    // 线程池默认线程数
    static const int gnum = 5;

    template <typename T>
    class ThreadPool
    {
    private:
        // 唤醒所有线程
        void WakeUpAllThread()
        {
            LockGuard lockguard(_mutex);                 // 加锁
            if (_sleepernum)                             // 如果有线程在等待
                _cond.Broadcast();                       // 唤醒所有等待的线程
            LOG(LogLevel::INFO) << "唤醒所有的休眠线程"; // 记录日志
        }

        // 唤醒一个线程
        void WakeUpOne()
        {
            _cond.Signal();                            // 唤醒一个等待的线程
            LOG(LogLevel::INFO) << "唤醒一个休眠线程"; // 记录日志
        }

        // 构造函数
        ThreadPool(int num = gnum) : _num(num), _isrunning(false), _sleepernum(0)
        {
            // 创建指定数量的线程
            for (int i = 0; i < num; i++)
            {
                _threads.emplace_back(
                    [this]() // 捕获this指针
                    {
                        HandlerTask(); // 线程执行任务处理函数
                    });
            }
        }

        // 启动线程池
        void Start()
        {
            if (_isrunning) // 如果线程池已经在运行
                return;
            _isrunning = true; // 设置线程池为运行状态
            // 启动所有线程
            for (auto &thread : _threads)
            {
                thread.Start();                                                       // 启动线程
                LOG(LogLevel::INFO) << "start new thread success: " << thread.Name(); // 记录日志
            }
        }

        // 禁用拷贝构造和赋值操作
        ThreadPool(const ThreadPool<T> &) = delete;
        ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;
    public:
        // 单例模式
        static ThreadPool<T>* GetInstance() {
            if(inc == nullptr) {
                LockGuard lockguard(_lock);
                LOG(LogLevel::DEBUG) << "获取单例...";
                if(inc == nullptr) {
                    LOG(LogLevel::DEBUG) << "首次使用单例，正在创建...";
                    inc = new ThreadPool<T>();
                    inc->Start();
                }
            }
            return inc;
        }
        // 停止线程池
        void Stop()
        {
            if (!_isrunning) // 如果线程池已经停止
                return;
            _isrunning = false; // 设置线程池为停止状态

            // 唤醒所有线程，使它们退出
            WakeUpAllThread();
        }

        // 等待所有线程结束
        void Join()
        {
            for (auto &thread : _threads)
            {
                thread.Join(); // 等待线程结束
            }
        }

        // 线程任务处理函数
        void HandlerTask()
        {
            char name[128];
            pthread_getname_np(pthread_self(), name, sizeof(name)); // 获取线程名称
            while (true)
            {
                T t;
                {
                    LockGuard lockguard(_mutex); // 加锁
                    // 如果任务队列为空且线程池未停止，等待任务
                    while (_taskq.empty() && _isrunning)
                    {
                        _sleepernum++;      // 增加等待线程数
                        _cond.Wait(_mutex); // 等待条件变量
                        _sleepernum--;      // 减少等待线程数
                    }
                    // 如果线程池已停止且任务队列为空，退出
                    if (!_isrunning && _taskq.empty())
                    {
                        LOG(LogLevel::INFO) << name << " 退出了, 线程池退出&&任务队列为空"; // 记录日志
                        break;
                    }

                    // 获取任务
                    t = _taskq.front(); // 从队列中获取任务
                    _taskq.pop();       // 移除任务
                }
                t(); // 执行任务
            }
        }

        // 添加任务到队列
        bool Enqueue(const T &in)
        {
            if (_isrunning) // 如果线程池在运行
            {
                LockGuard lockguard(_mutex); // 加锁
                _taskq.push(in);             // 添加任务到队列
                // 如果所有线程都在等待，唤醒一个线程
                if (_threads.size() == _sleepernum)
                    WakeUpOne();
                return true;
            }
            return false;
        }

        // 析构函数
        ~ThreadPool()
        {
            // 在析构时停止线程池并等待线程结束
            Stop();
            Join();
        }

    private:
        std::vector<Thread> _threads; // 线程池中的线程
        int _num;                     // 线程池中线程的数量
        std::queue<T> _taskq;         // 任务队列
        Cond _cond;                   // 条件变量
        Mutex _mutex;                 // 互斥锁
        bool _isrunning;              // 线程池是否在运行
        int _sleepernum;              // 等待的线程数

        // 懒汉方式
        static ThreadPool<T>* inc;
        static Mutex _lock;
    };

    // 单例指针的初始化
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::inc = nullptr;

    // 单例锁的初始化
    template <typename T>
    Mutex ThreadPool<T>::_lock;
}